# 一方库

from .all import Dataset_siamese,Model_siamese,Loss_siamese


# 三方库
from torch.utils.data import DataLoader

def get_param_siamese(dataset_one,model_one,loss_one,batch_size,device):
    """获得siamese网络的所有实体"""
    

    dataset_siamese    = Dataset_siamese(dataset_one)
    dataLoader_siamese = DataLoader(dataset=dataset_siamese,batch_size=batch_size,drop_last=True,shuffle=False)
    model_siamese      = Model_siamese(model_one).to(device)
    loss_siamese       = Loss_siamese(loss_one,margin=8)
    return (dataset_siamese,dataLoader_siamese,model_siamese,loss_siamese,model_one)